import sys
import os
sys.path.append('')
from GNNSwitch.graph import *

if __name__ == '__main__':

   # MaskedData = [
   #    "Reddit",
   #    "FacebookPagePage"
   # ]
   # MaskedData += ["GNNAdvData."+x for x in os.listdir("data/GNNAdvData/processed")]
   # MaskedData += ["Twitch."+x for x in os.listdir("data/Twitch")]
   # MaskedData += ["Planetoid."+x for x in os.listdir("data/Planetoid")]
   # MaskedData += ["SNAPDataset."+x for x in os.listdir("data/SNAPDataset")]
   # MaskedData += ["PPI"]
   MaskedData = [ "Mtx." + x for x in os.listdir("data/Mtx/raw") ]
   header = True
   for d in MaskedData:
      sweep_dataset(d,header)
      header = False

   # sname = 'Reddit' if len(sys.argv) < 2 else sys.argv[1]
   # data = sweep_dataset(sname)
   # data[3].toGraphViz("dot.gv")